package com.mango.repository;

import com.mango.facade.param.TemplateAttribute;
import com.mango.repository.convert.Convert;
import com.mango.repository.mapper.AmazonTemplateAttributeMapper;
import com.mango.repository.model.AmazonTemplateAttribute;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

@Repository
public class TemplateAttributeRepository {
    @Autowired
    private AmazonTemplateAttributeMapper amazonTemplateAttributeMapper;


    public List<TemplateAttribute> getAttributeList(String templateName) {
        ArrayList<String> arrayList = new ArrayList<>();
        List<AmazonTemplateAttribute> attributeList = amazonTemplateAttributeMapper.selectByPrimaryTemplateName(arrayList);
        return attributeList.stream().map(Convert::toTemplateAttribute).collect(Collectors.toList());
    }

    public List<TemplateAttribute> getAttributeByList(List<String> listName) {
        List<AmazonTemplateAttribute> attributeList = amazonTemplateAttributeMapper.selectByPrimaryTemplateName(listName);
        return attributeList.stream().map(Convert::toTemplateAttribute).collect(Collectors.toList());
    }
}
